import argparse
from train_Mal import train_model


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    default_str = 'default: %(default)s'
    parser.add_argument('name', type=str)
    parser.add_argument('--valid_folds', type=int, nargs='*', default=[0, ], help=default_str)
    parser.add_argument('--batchsize', type=int, default=16, help=default_str)
    parser.add_argument('--shift_range', type=int, nargs=3, default=[2, 2, 2], help=default_str)
    parser.add_argument('--crop_shape', type=int, nargs=3, default=[64, 64, 64], help=default_str)
    parser.add_argument('--seed', type=int, default=2112, help=default_str)
    parser.add_argument('--lr', type=float, default=1.0, help=default_str)
    parser.add_argument('--lr_decay', type=float, default=1.0, help=default_str)
    parser.add_argument('--nb_epoches', type=int, default=35, help=default_str)
    parser.add_argument('--weight_decay', type=float, default=3e-4, help=default_str)
    parser.add_argument('--gpus', type=str, default='0', help=default_str)
    args = parser.parse_args()
    train_model(**vars(args))
